package jnpf.model.ragFlow.common;


import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;
import jnpf.model.ragFlow.file.ParserConfig;
import lombok.Data;

@Data
@ApiModel("数据集信息")
public class Dataset {
    @ApiModelProperty("头像的Base64编码")
    private String avatar;

    @ApiModelProperty("分块数量")
    private Integer chunkCount;

    @ApiModelProperty("分块方法")
    private String chunkMethod = "General";

    @ApiModelProperty("创建日期")
    private String createDate;

    @ApiModelProperty("创建时间戳")
    private Long createTime;

    @ApiModelProperty("创建者ID")
    private String createdBy;

    @ApiModelProperty("数据集描述")
    private String description;

    @ApiModelProperty("文档数量")
    private Integer documentCount;

    @ApiModelProperty("嵌入模型名称")
    private String embeddingModel = "BAAI/bge-large-zh-v1.5";

    @ApiModelProperty("知识库ID")
    private String id;

    @ApiModelProperty("语言")
    private String language;

    @ApiModelProperty("数据集名称")
    private String name;

    @ApiModelProperty("页面排名")
    private Integer pagerank = 0;

    @ApiModelProperty("解析器配置")
    private ParserConfig parserConfig = new ParserConfig();

    @ApiModelProperty("访问权限")
    private String permission = "me";

    @ApiModelProperty("相似度阈值")
    private Double similarityThreshold;

    @ApiModelProperty("状态")
    private String status;

    @ApiModelProperty("租户ID")
    private String tenantId;

    @ApiModelProperty("token数量")
    private Integer tokenNum = 512;

    @ApiModelProperty("更新日期")
    private String updateDate;

    @ApiModelProperty("更新时间戳")
    private Long updateTime;

    @ApiModelProperty("向量相似度权重")
    private Double vectorSimilarityWeight;
}
